from ultralytics import YOLO

# 全新训练
# model = YOLO('yolov8n.yaml')

# 加载预训练模型
model = YOLO('yolov8n.pt')

# 全新训练，转移预训练模型的权重
# model = YOLO('yolov8n.yaml').load('yolov8n.pt')

# 开始训练
# results = model.train(data='coco128.yaml', epochs=100, imgsz=640)
